from predict import ChurnPrediction
from train import ChurnModel

def main():
    cm = ChurnModel('data/train.csv')
    cm.model_train()
    cm.model_save()

    cp= ChurnPrediction('data/test.csv',cm.model_path,cm.scalar,cm.encoder_dict)
    cp.prediction()

if __name__ == '__main__':
    main()
